import unittest

import testbaidu2
import testbaidu1


def creatSuit():
    # 要把不同的测试脚本的类中的需要执行的方法放在一个测试套件中
    # suit = unittest.TestSuite()
    # suit.addTest(testbaidu1.Baidu1("test_hao"))
    # suit.addTest(testbaidu2.Baidu2("test_hao"))
    # suit.addTest(testbaidu2.Baidu2("test_baidusearch"))
    # return suit

    # 如果我需要把一个测试脚本中年所有的测试用例都添加到suit中，怎么做？
    # makeSuit
    #  suit = unittest.TestSuite()
    #  suit.addTest(unittest.makeSuite(testbaidu1.Baidu1))
    #  suit.addTest(unittest.makeSuite(testbaidu2.Baidu2))
    #  return suit

    #     TestLoader
    #     suit1 = unittest.TestLoader().loadTestsFromTestCase(testbaidu1.Baidu1)
    #     suit2 = unittest.TestLoader().loadTestsFromTestCase(testbaidu2.Baidu2)
    #     suit = unittest.TestSuite([suit1, suit2])
    #     return suit
    # 可以把一个文件夹下面所有的测试脚本中的测试用例放入测试套件
    discover = unittest.defaultTestLoader.discover("", pattern="testbaidu*.py", top_level_dir=None)
    return discover


if __name__ == "__main__":
    suit = creatSuit()
    # verbersity= 0, 1, 2
    runner = unittest.TextTestRunner()
    runner.run(suit)
